import type { Router } from 'vue-router'
import NProgress from 'nprogress'
import { useUserStore } from '@/stores/modules/user'
import { usePermissionStore } from '@/stores/modules/permission'
import { getToken } from '@/utils/auth'
import { ElMessage } from 'element-plus'

// 白名单路由
const whiteList = ['/login', '/404', '/401']

/**
 * 设置路由守卫
 * @param router 路由实例
 */
export function setupRouterGuard(router: Router) {
  // 全局前置守卫
  router.beforeEach(async (to, from, next) => {
    // 开始进度条
    NProgress.start()

    // 设置页面标题
    document.title = getPageTitle(to.meta?.title as string)

    const hasToken = getToken()

    if (hasToken) {
      if (to.path === '/login') {
        // 如果已登录，重定向到首页
        next({ path: '/' })
        NProgress.done()
      } else {
        const userStore = useUserStore()
        const hasRoles = userStore.roles && userStore.roles.length > 0

        if (hasRoles) {
          next()
        } else {
          try {
            // 获取用户信息
            await userStore.getUserInfoAction()

            // 生成可访问的路由表
            const permissionStore = usePermissionStore()
            const accessRoutes = await permissionStore.generateRoutes(userStore.roles)

            // 动态添加可访问路由
            accessRoutes.forEach(route => {
              router.addRoute(route)
            })

            // hack方法 确保addRoutes已完成
            next({ ...to, replace: true })
          } catch (error) {
            // 移除token并跳转登录页
            await userStore.resetUserInfo()
            ElMessage.error('登录状态已过期，请重新登录')
            next(`/login?redirect=${to.path}`)
            NProgress.done()
          }
        }
      }
    } else {
      // 没有token
      if (whiteList.indexOf(to.path) !== -1) {
        // 在免登录白名单，直接进入
        next()
      } else {
        // 其他没有访问权限的页面将被重定向到登录页面
        next(`/login?redirect=${to.path}`)
        NProgress.done()
      }
    }
  })

  // 全局后置守卫
  router.afterEach(() => {
    // 结束进度条
    NProgress.done()
  })

  // 全局错误处理
  router.onError(error => {
    console.error('路由错误:', error)
    NProgress.done()
  })
}

/**
 * 获取页面标题
 * @param pageTitle 页面标题
 * @returns 完整标题
 */
function getPageTitle(pageTitle?: string): string {
  const title = 'Kexilo权限管理系统'
  if (pageTitle) {
    return `${pageTitle} - ${title}`
  }
  return title
}
